package cn.aposoft.mq.kafka.admin.consumer;

import org.apache.kafka.clients.admin.Admin;
import org.apache.kafka.clients.admin.ConsumerGroupListing;
import org.apache.kafka.clients.admin.ListConsumerGroupsResult;
import org.apache.kafka.clients.admin.NewTopic;
import org.apache.kafka.clients.producer.ProducerConfig;
import org.apache.kafka.common.KafkaFuture;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Properties;
import java.util.concurrent.ExecutionException;

public class ListConsumerGroups {

    public static void main(String[] args) {
        final String topicName = "test-add-partition";
        Properties p = new Properties();
        p.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "aposoft.cn:10091");// kafka地址，多个地址用逗号分割
        try (Admin admin = Admin.create(p)) {
            List<NewTopic> topics = new ArrayList<>();
            NewTopic topic = new NewTopic(topicName, 1, (short) 1);
            topics.add(topic);
            ListConsumerGroupsResult result = admin.listConsumerGroups();
            KafkaFuture<Collection<ConsumerGroupListing>> allFuture = result.all();
            Collection<ConsumerGroupListing> all = allFuture.get();
            System.out.println(all);
            KafkaFuture<Collection<ConsumerGroupListing>> validFuture = result.valid();
            System.out.println(validFuture.get());
        } catch (ExecutionException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

}
